import prompt
import api

class BaseAgent:
    def __init__(self, llm: api.ZhipuLLM, system_prompt: str):
        self.llm = llm
        self.system_prompt = system_prompt
        self.memory = []

    def query(self, prompt: str):
        return self.llm.query_with_system(prompt, self.system_prompt)

    

class FindAgent(BaseAgent):
    def __init__(self, llm: api.ZhipuLLM):
        super().__init__(llm, prompt.role_info["finder"])
        self.paper_search_api = api.LiteratureSearchAPIAsync(config.paper_search_api_url)

    


if __name__ == "__main__":
    import config

    llm = api.ZhipuLLM(config.zhipu_api_key)
    agent = FindAgent(llm)
    agent.query("Who are you?")